package com.zhu.linkedlist;


/**
 * 链表
 *
 * @author: zhu
 * @date: 2019/1/13 10:34
 */
public class MyLinkedList<E> {

    /** 节点设计成内部类 */
    private class Node {
        public E e;//存放的元素
        public Node next;//下一个节点
        public Node(E e, Node next) {
            this.e = e;
            this.next = next;
        }
        public Node(E e) {
            this(e, null);
        }
        public Node() {
            this(null, null);
        }
        @Override
        public String toString() {
            return e.toString();
        }
    }
    private Node dummyhead;//虚拟头节点
    private int size;//存储的元素个数

    public MyLinkedList() {
        dummyhead = new Node(null, null);
        size = 0;
    }

    /** 获取链表中元素个数 */
    public int getSize() {
        return size;
    }

    /** 判断链表是否为空 */
    public boolean isEmpty() {
        return size == 0;
    }

    /** 在链表指定位置添加元素，(index是从0开始的) */
    public void add(int index, E e) {
        if (index < 0 || index > size) {
            throw new IllegalArgumentException("位置不合法");
        }
        Node prev = dummyhead;
        for (int i = 0; i < index; i++) {
            prev = prev.next;//找到目标位置的前一个节点
        }
            /*Node node = new Node(e);
            node.next = prev.next;
            prev.next = node;*/
        prev.next = new Node(e, prev.next);//与上面3行代码等效
        size++;
    }

    /** 在链表末尾添加元素 */
    public void addLast(E e) {
        add(size, e);
    }

    /** 往链表头添加元素 */
    public void addFirst(E e) {
        add(0,e);
    }

    /** 获取链表指定位置的元素(index从0开始) */
    public E get(int index){
        if (index<0 || index>=size){
            throw new IllegalArgumentException("位置不合法");
        }
        Node cur = dummyhead.next;
        for (int i=0;i<index;i++){
            cur = cur.next;
        }
        return cur.e;
    }

    /** 获取第一个元素 */
    public E getFirst(){
        return get(0);
    }

    /** 获取链表最后一个元素 */
    public E getLast(){
        return get(size -1);
    }

    /** 更新指定位置的元素 */
    public void set(int index,E e){
        if (index<0 || index>=size){
            throw new IllegalArgumentException("位置不合法");
        }
        Node cur = dummyhead.next;
        for (int i=0;i<index;i++){
            cur = cur.next;
        }
        cur.e = e;
    }

    /** 查找链表中是否存在元素e */
    public boolean contains(E e){
        Node cur = dummyhead.next;
        while (cur != null){
            if (cur.e.equals(e))
                return true;
            cur = cur.next;
        }
        return false;
    }

    /** 删除指定位置的元素 */
    public E remove(int index){
        if (index<0 || index>=size){
            throw new IllegalArgumentException("位置不合法");
        }
        Node prev = dummyhead;
        for (int i = 0;i<index;i++){
            prev = prev.next;//找到目标节点的前一个节点
        }
        Node delNode = prev.next;
        prev.next = delNode.next;
        delNode.next = null;
        size --;
        return delNode.e;
    }

    /** 删除指定元素e */
    public E remove(E e){
        Node node = dummyhead;
        int index = -1;
        while (node.e != e){
            node = node.next;
            index ++;
        }
        return remove(index);
    }
    /** 删除链表中第一个元素 */
    public E removeFirst(){
        return remove(0);
    }

    /** 删除链表中最后一个元素 */
    public E removeLast(){
        return remove(size -1);
    }

    @Override
    public String toString(){
        StringBuilder res = new StringBuilder();
        Node cur = dummyhead.next;
        while (cur != null){
            res.append(cur + "--> ");
            cur = cur.next;
        }
        res.append("NULL");
        return res.toString();
    }
}
